package com.chj.leetcode.tijie300to350;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.PriorityQueue;

public class Code347_TopKFrequentElements {
	public int[] topKFrequent(int[] nums, int k) {
		// 使用字典，统计每个元素出现的次数，元素为键，元素出现的次数为值
		HashMap<Integer, Integer> map = new HashMap<>();
		for (int num : nums) {
			if (map.containsKey(num)) {
				map.put(num, map.get(num) + 1);
			} else {
				map.put(num, 1);
			}
		}

		// 遍历map，用最小堆保存频率最大的k个元素
		PriorityQueue<Integer> pq = new PriorityQueue<>(new Comparator<Integer>() {
			@Override
			public int compare(Integer a, Integer b) {
				return map.get(a) - map.get(b);
			}
		});

		for (Integer key : map.keySet()) {
			if (pq.size() < k) {
				pq.add(key);
			} else if (map.get(key) > map.get(pq.peek())) {
				pq.remove();
				pq.add(key);
			}
		}

		// 取出最小堆中的元素
		int[] res = new int[k];
		int index = 0;
		while (!pq.isEmpty()) {
			res[index++] = pq.remove();
		}
		return res;
	}

//	作者：MisterBooo
//	链接：https://leetcode-cn.com/problems/top-k-frequent-elements/solution/leetcode-di-347-hao-wen-ti-qian-k-ge-gao-pin-yuan-/
//	来源：力扣（LeetCode）
//	著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

	public List<String> topKFrequent(String[] words, int k) {
		// 使用字典，统计每个元素出现的次数，元素为键，元素出现的次数为值
		HashMap<String, Integer> map = new HashMap<>();
		for (String num : words) {
			if (map.containsKey(num)) {
				map.put(num, map.get(num) + 1);
			} else {
				map.put(num, 1);
			}
		}

		// 遍历map，用最小堆保存频率最大的k个元素
		PriorityQueue<String> pq = new PriorityQueue<>(new Comparator<String>() {
			@Override
			public int compare(String a, String b) {
				if (map.get(a) - map.get(b) == 0) {
					return ((b).compareTo(a));
				}
				return map.get(a) - map.get(b);
			}
		});

		for (String key : map.keySet()) {
			if (pq.size() < k) {
				pq.offer(key);

			} else if (map.get(key) == map.get(pq.peek())) {
				if ((key).compareTo(pq.peek()) < 0) {
					pq.poll();
					pq.offer(key);
				}

			} else if (map.get(key) > map.get(pq.peek())) {
				pq.poll();
				pq.offer(key);
			}
		}

		List<String> res = new ArrayList<>();
		while (!pq.isEmpty()) {
			res.add(pq.poll());
		}

		Collections.sort(res,
				(o1, o2) -> (map.get(o2) - map.get(o1) == 0 ? o1.compareTo(o2) : map.get(o2) - map.get(o1)));
		return res;
	}

//	作者：LeetCode
//	链接：https://leetcode-cn.com/problems/top-k-frequent-words/solution/qian-kge-gao-pin-dan-ci-by-leetcode/
//	来源：力扣（LeetCode）
//	著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

	/**
	 * 1. 统计词频 2. 构建最小堆 3. 弹出答案
	 */
	public static List<String> topKFrequent03(String[] words, int k) {
		// 1. 构建词频
		HashMap<String, Integer> map = new HashMap<>();
		for (String word : words)
			map.put(word, map.getOrDefault(word, 0) + 1);
		// 2. 构建最小堆
		PriorityQueue<String> minHeap = new PriorityQueue<>(
				(o1, o2) -> (map.get(o1) - map.get(o2) == 0 ? ((String) o2).compareTo((String) o1)
						: map.get(o1) - map.get(o2)));
		for (String word : map.keySet()) {
//			minHeap.add(word);
//			if (minHeap.size() > k)
//				minHeap.poll();

			if (minHeap.size() < k) {
				minHeap.add(word);
			} else if (map.get(word) > map.get(minHeap.peek())) {
				minHeap.poll();
				minHeap.add(word);
			}
		}
		// 3. 弹出答案
		List<String> ans = new ArrayList<>();
		while (!minHeap.isEmpty()) {
			ans.add(minHeap.poll());
		}
		Collections.sort(ans,
				(o1, o2) -> (map.get(o2) - map.get(o1) == 0 ? o1.compareTo(o2) : map.get(o2) - map.get(o1)));
		return ans;
	}

//作者：user7692w
//链接：https://leetcode-cn.com/problems/top-k-frequent-words/solution/topkwen-ti-jie-ti-si-lu-san-bu-qu-by-user7692w/
//来源：力扣（LeetCode）
//著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

}
